package com.leetcode.leetcode3;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 找到字符串中所有字母异位词
 *
 * @author luhd
 * @date 2023-08-09
 */
public class Practice438 {
    /**
     * 给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。
     *
     * 异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。
     * 输入: s = "cbaebabacd", p = "abc"
     * 输出: [0,6]
     * 解释:
     * 起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
     * 起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。
     */
    public static void main(String[] args) {
        // String s = "cbaebabacd";
        // String p = "abc";
        String s = "abab";
        String p = "ab";
        System.out.println(findAnagrams(s, p));
    }

    public static List<Integer> findAnagrams(String s, String p) {
        /*char[] chars = p.toCharArray();
        Arrays.sort(chars);
        p = new String(chars);
        int n1 = s.length();
        int n2 = p.length();
        List<Integer> list = new ArrayList<>();
        if (n1<n2) {
            return list;
        }
        for (int i = 0; i <= n1 - n2; i++) {
            String sub = s.substring(i, i + n2);
            char[] chars1 = sub.toCharArray();
            Arrays.sort(chars1);
            sub = new String(chars1);
            if (sub.equals(p)) {
                list.add(i);
            }
        }
        return list;*/
        int sLen = s.length(), pLen = p.length();

        if (sLen < pLen) {
            return new ArrayList<Integer>();
        }

        List<Integer> ans = new ArrayList<Integer>();
        int[] sCount = new int[26];
        int[] pCount = new int[26];
        for (int i = 0; i < pLen; ++i) {
            ++sCount[s.charAt(i) - 'a'];
            ++pCount[p.charAt(i) - 'a'];
        }

        if (Arrays.equals(sCount, pCount)) {
            ans.add(0);
        }

        for (int i = 0; i < sLen - pLen; ++i) {
            --sCount[s.charAt(i) - 'a'];
            ++sCount[s.charAt(i + pLen) - 'a'];

            if (Arrays.equals(sCount, pCount)) {
                ans.add(i + 1);
            }
        }

        return ans;
    }
}